import pandas as pd

# 读取两个 Excel 文件
excel1_path = './data/file1.xlsx'  # 替换为你的 excel1 文件路径
excel2_path = './data/base/file2.xlsx'  # 替换为你的 excel2 文件路径

# 假设 Excel 文件中的列名如下：
# excel1 中的公司列名为 '公司名称'
# excel2 中的公司列名为 '公司名称'，产品服务列名为 '产品服务'

# 读取 excel1 和 excel2
df_excel1 = pd.read_excel(excel1_path)
df_excel2 = pd.read_excel(excel2_path)

# 确保列名正确
company_col_excel1 = 'company_name'
company_col_excel2 = 'company_name'
product_service_col = 'Products_and_Services'

# 创建一个新列用于存储匹配的产品服务信息
df_excel1['Products_and_Services'] = None

# 遍历 excel1 的每一行，查找是否在 excel2 中存在对应的公司
for index, row in df_excel1.iterrows():
    company_name = row[company_col_excel1]
    # 在 excel2 中查找是否存在该公司
    match_row = df_excel2[df_excel2[company_col_excel2] == company_name]
    if not match_row.empty:
        # 如果找到匹配项，提取产品服务并添加到 excel1 对应的行
        df_excel1.at[index, 'Products_and_Services'] = match_row.iloc[0][product_service_col]

# 将更新后的 excel1 保存到新的 Excel 文件
output_path = 'updated_excel1.xlsx'
df_excel1.to_excel(output_path, index=False)

print(f"处理完成，结果已保存到 {output_path}")